#!/usr/bin/python
#coding:utf-8
'''
图像风格转换V1算法实现
'''
import numpy as np
vgg16_data=np.load('./vgg16.npy',allow_pickle=True,encoding='bytes') #加载VGGnet16预训练模型
print(type(vgg16_data)) #查看数据类型
data_dict= vgg16_data.item() #转换为字典
print(data_dict.keys()) #获取所有key
print(len(data_dict))

#查看卷积层
conv1_1 = data_dict[b'conv1_1']
# pring(len(conv1_1))
w,b =conv1_1 #w 卷积核参数，b 偏置
print(w.shape)
print(b.shape)

#查看全连接层
fc6 = data_dict[b'fc6']
print(len(fc6))

w,b =fc6
print(w.shape)
print(b.shape)